<?php
// +----------------------------------------------------------------------
// | zip压缩包操作类
// +----------------------------------------------------------------------
namespace app\common\lib;

class ZipFolder
{
    protected $zip;
    protected $root;
    protected $ignored_names;

    public function __construct()
    {
        $this->zip = new ZipArchive;
    }

    /**
     * 解压zip文件到指定文件夹
     *
     * @access public
     * @param string $zipfile 压缩文件路径
     * @param string $path 压缩包解压到的目标路径
     * @return booleam 解压成功返回 true 否则返回 false
     */
    public function unzip($zipfile, $path)
    {
        if ($this->zip->open($zipfile) === true) {
            $file_tmp = @fopen($zipfile, "rb");
            $bin = fread($file_tmp, 15); //只读15字节 各个不同文件类型，头信息不一样。
            fclose($file_tmp);
            /* 只针对zip的压缩包进行处理 */
            if (true === $this->getTypeList($bin)) {
                $result = $this->zip->extractTo($path);
                $this->zip->close();
                return $result;
            } else {
                return false;
            }
        }
        return false;
    }

    /**
     * 创建压缩文件
     *
     * @access public
     * @param string $zipfile 将要生成的压缩文件路径
     * @param strng $folder 将要被压缩的文件夹路径
     * @param array $ignored 要忽略的文件列表
     * @return booleam 压缩包生成成功返回true 否则返回 false
     */
    public function zip($zipfile, $folder, $ignored = null)
    {
        $this->ignored_names = is_array($ignored) ? $ignored : $ignored ? array($ignored) : array();
        if ($this->zip->open($zipfile, ZIPARCHIVE::CREATE) !== true) {
            throw new Exception("cannot open <$zipfile>\n");
        }
        $folder = substr($folder, -1) == '/' ? substr($folder, 0, strlen($folder) - 1) : $folder;
        if (strstr($folder, '/')) {
            $this->root = substr($folder, 0, strrpos($folder, '/') + 1);
            $folder = substr($folder, strrpos($folder, '/') + 1);
        }
        $this->createZip($folder);
        return $this->zip->close();
    }

    /**
     * 递归添加文件到压缩包
     *
     * @access private
     * @param string $folder 添加到压缩包的文件夹路径
     * @param string $parent 添加到压缩包的文件夹上级路径
     * @return void
     */
    private function createZip($folder, $parent = null)
    {
        $full_path = $this->root . $parent . $folder;
        $zip_path = $parent . $folder;
        $this->zip->addEmptyDir($zip_path);
        $dir = new DirectoryIterator($full_path);
        foreach ($dir as $file) {
            if (!$file->isDot()) {
                $filename = $file->getFilename();
                if (!in_array($filename, $this->ignored_names)) {
                    if ($file->isDir()) {
                        $this->createZip($filename, $zip_path . '/');
                    } else {
                        $this->zip->addFile($full_path . '/' . $filename, $zip_path . '/' . $filename);
                    }
                }
            }
        }
    }

    /**
     * 读取压缩包文件与目录列表
     *
     * @access public
     * @param string $zipfile 压缩包文件
     * @return array 文件与目录列表
     */
    public function fileList($zipfile)
    {
        $file_dir_list = array();
        $file_list = array();
        if ($this->zip->open($zipfile) == true) {
            for ($i = 0; $i < $this->zip->numFiles; $i++) {
                $numfiles = $this->zip->getNameIndex($i);
                if (preg_match('/\/$/i', $numfiles)) {
                    $file_dir_list[] = $numfiles;
                } else {
                    $file_list[] = $numfiles;
                }
            }
        }
        return array('files' => $file_list, 'dirs' => $file_dir_list);
    }

    /**
     * 得到文件头与文件类型映射表
     *
     * @param $bin string 文件的二进制前一段字符
     * @return boolean
     */
    private function getTypeList($bin)
    {
        $array = array(
            array("504B0304", "zip")
        );
        foreach ($array as $v) {
            $blen = strlen(pack("H*", $v[0])); //得到文件头标记字节数
            $tbin = substr($bin, 0, intval($blen)); ///需要比较文件头长度
            if (strtolower($v[0]) == strtolower(array_shift(unpack("H*", $tbin)))) {
                return true;
            }
        }
        return false;
    }
	
	/**
     * Add files and sub-directories in a folder to zip file.
     * @param string $folder
     * @param ZipArchive $zipFile
     * @param int $exclusiveLength Number of text to be exclusived from the file path.
     */
    public function folderToZip($folder, &$zipFile, $exclusiveLength)
    {
        $handle = opendir($folder);
        while (false !== $f = readdir($handle)) {
            if ($f != '.' && $f != '..') {
                $filePath = "$folder/$f";
                // Remove prefix from file path before add to zip.
                $localPath = substr($filePath, $exclusiveLength);
                if (is_file($filePath)) {
                    $zipFile->addFile($filePath, $localPath);
                } elseif (is_dir($filePath)) {
                    // Add sub-directory.
                    $zipFile->addEmptyDir($localPath);
                    $this->folderToZip($filePath, $zipFile, $exclusiveLength);
                }
            }
        }
        @closedir($handle);
    }

    /**
     * zip压缩[多级]目录，不支持中文文件/目录
     * Zip a folder (include itself).
     * Usage:
     *   zipDir('/path/to/sourceDir', '/path/to/out.zip');
     *
     * @param string $sourcePath Path of directory to be zip.
     * @param string $outZipPath Path of output zip file.
     */
    public function zipDir($sourcePath, $outZipPath)
    {
        $pathInfo = pathInfo($sourcePath);
        $parentPath = $pathInfo['dirname'];
        $dirName = $pathInfo['basename'];

        $z = new \ZipArchive();
        $z->open($outZipPath, \ZipArchive::CREATE);
        $z->addEmptyDir($dirName);
        $this->folderToZip($sourcePath, $z, strlen("$parentPath/"));
        $z->close();
    }

	/**
     * zip压缩支持中文目录/文件名
	 * @param array $files file_path要压缩的文件，绝对路径 file_name压缩后的文件含目录层级
	 * @param string $create_zip_absolute_path 生成的zip文件，绝对路径
     */
	public function zip_add_from_tring($files, $create_zip_absolute_path)
    {
        /* $files = array(
            array("file_path" => $_SERVER['DOCUMENT_ROOT'] . "/uploads/666.jpg", "file_name" => "123.jpg"),
            array("file_path" => $_SERVER['DOCUMENT_ROOT'] . "/uploads/123.pdf", "file_name" => "测试/测试.pdf")
        );
		$create_zip_absolute_path = $_SERVER['DOCUMENT_ROOT'] . "/uploads/" . time() . ".zip"; //下载后的文件名 */

        $zip = new \ZipArchive(); // 新建一个ZipArchive的对象

        // 开始操作压缩包
        if ($zip->open($create_zip_absolute_path, \ZipArchive::CREATE) === TRUE) {
            foreach ($files as $k => $v) {
                $zip->addFromString(iconv('utf-8', 'gbk//ignore', $v['file_name']), file_get_contents($v['file_path'])); //向压缩包中添加文件
            }

            $zip->close(); //关闭压缩包
        }
	}
}